JavaScript 迭代器助手流管理器：流处理系统

            
const numbers = [1, 2, 3, 4, 5, 6, 7, 8, 9, 10];

const squaredOddNumbers = numbers
  .filter(number => number % 2 !== 0)
  .map(number => number * number);

console.log(squaredOddNumbers); // Output: [1, 9, 25, 49, 81]

            

              
                Copy
              
            
          

            
class StreamManager {
  constructor() {
    this.source = null;
    this.transformations = [];
    this.destination = null;
    this.errorHandler = null;
  }

  setSource(source) {
    this.source = source;
    return this;
  }

  addTransformation(transformation) {
    this.transformations.push(transformation);
    return this;
  }

  setDestination(destination) {
    this.destination = destination;
    return this;
  }

  setErrorHandler(errorHandler) {
    this.errorHandler = errorHandler;
    return this;
  }

  async *process() {
    if (!this.source) {
      throw new Error("Source not defined");
    }

    try {
      for await (const data of this.source) {
        let transformedData = data;
        for (const transformation of this.transformations) {
          transformedData = await transformation(transformedData);
        }
        yield transformedData;
      }
    } catch (error) {
      if (this.errorHandler) {
        this.errorHandler(error);
      } else {
        console.error("Error processing stream:", error);
      }
    }
  }

  async run() {
    if (!this.destination) {
      throw new Error("Destination not defined");
    }

    try {
      for await (const data of this.process()) {
        await this.destination(data);
      }
    } catch (error) {
      console.error("Error running stream:", error);
    }
  }
}

// Example usage:
async function* generateNumbers(count) {
  for (let i = 0; i < count; i++) {
    yield i;
    await new Promise(resolve => setTimeout(resolve, 100)); // Simulate delay
  }
}

async function squareNumber(number) {
  return number * number;
}

async function logNumber(number) {
  console.log("Processed:", number);
}

const streamManager = new StreamManager();

streamManager
  .setSource(generateNumbers(10))
  .addTransformation(squareNumber)
  .setDestination(logNumber)
  .setErrorHandler(error => console.error("Custom error handler:", error));

streamManager.run();

            

              
                Copy
              
            
          

            
const fs = require('fs');
const readline = require('readline');

async function* readFileLines(filePath) {
  const fileStream = fs.createReadStream(filePath);

  const rl = readline.createInterface({
    input: fileStream,
    crlfDelay: Infinity
  });

  for await (const line of rl) {
    yield line;
  }
}

// Example usage:
streamManager.setSource(readFileLines('data.txt'));

            

              
                Copy
              
            
          

            
async function* fetchAPI(url) {
  let page = 1;
  while (true) {
    const response = await fetch(`${url}?page=${page}`);
    const data = await response.json();

    if (!data || data.length === 0) {
      break; // No more data
    }

    for (const item of data) {
      yield item;
    }

    page++;
    await new Promise(resolve => setTimeout(resolve, 500)); // Rate limiting
  }
}

// Example usage:
streamManager.setSource(fetchAPI('https://api.example.com/data'));

            

              
                Copy
              
            
          

            
const { Kafka } = require('kafkajs');

async function* consumeKafka(topic, groupId) {
  const kafka = new Kafka({
    clientId: 'my-app',
    brokers: ['localhost:9092']
  });

  const consumer = kafka.consumer({ groupId: groupId });
  await consumer.connect();
  await consumer.subscribe({ topic: topic, fromBeginning: true });

  await consumer.run({
    eachMessage: async ({ message }) => {
      yield message.value.toString();
    },
  });

  // Note: Consumer should be disconnected when stream is finished.
  // For simplicity, disconnection logic is omitted here.
}

// Example usage:
// Note: Ensure Kafka broker is running and topic exists.
// streamManager.setSource(consumeKafka('my-topic', 'my-group'));

            

              
                Copy
              
            
          

            
async function enrichWithUserData(data) {
  // Assume we have a function to fetch user data by ID
  const userData = await fetchUserData(data.userId);
  return { ...data, user: userData };
}

// Example usage:
streamManager.addTransformation(enrichWithUserData);

            

              
                Copy
              
            
          

            
function filterByCountry(data, countryCode) {
  if (data.country === countryCode) {
    return data;
  }
  return null; // Or throw an error, depending on desired behavior
}

// Example usage:
streamManager.addTransformation(async (data) => filterByCountry(data, 'US'));

            

              
                Copy
              
            
          

            
async function aggregateData(data) {
    // Simple example: keeps a running count.
    aggregateData.count = (aggregateData.count || 0) + 1;
    return { ...data, count: aggregateData.count };
}

// Example usage
streamManager.addTransformation(aggregateData);


            

              
                Copy
              
            
          

            
const fs = require('fs');

async function writeToFile(data, filePath) {
  fs.appendFileSync(filePath, JSON.stringify(data) + '\n');
}

// Example usage:
streamManager.setDestination(async (data) => writeToFile(data, 'output.txt'));

            

              
                Copy
              
            
          

            
async function sendToAPI(data, apiUrl) {
  const response = await fetch(apiUrl, {
    method: 'POST',
    headers: {
      'Content-Type': 'application/json'
    },
    body: JSON.stringify(data)
  });

  if (!response.ok) {
    throw new Error(`API request failed: ${response.status}`);
  }
}

// Example usage:
streamManager.setDestination(async (data) => sendToAPI(data, 'https://api.example.com/results'));

            

              
                Copy
              
            
          

            
const { Kafka } = require('kafkajs');

async function publishToKafka(data, topic) {
  const kafka = new Kafka({
    clientId: 'my-app',
    brokers: ['localhost:9092']
  });

  const producer = kafka.producer();
  await producer.connect();

  await producer.send({
    topic: topic,
    messages: [
      {
        value: JSON.stringify(data)
      }
    ],
  });

  await producer.disconnect();
}

// Example usage:
// Note: Ensure Kafka broker is running and topic exists.
// streamManager.setDestination(async (data) => publishToKafka(data, 'my-output-topic'));

            

              
                Copy